"""
伯努利朴素贝叶斯光线类别分类
"""
from sklearn.naive_bayes import GaussianNB

from 光照.dataset import LightDataset

train_dataset = LightDataset()
X, y = train_dataset.get_data()

test_dataset = LightDataset(data_type="test")
test_X, test_y = test_dataset.get_data()

nb = GaussianNB()
nb.fit(X, y)

score = nb.score(test_X, test_y)
print(score)
